package config

import (
	"fmt"

	"github.com/spf13/viper"
)

var C *Config

type UserInfo struct {
	User string
	Pass string
}

type Config struct {
	Src   string
	Port  int64
	Users []UserInfo
}

func init() {
	viper.SetConfigFile("./config.yaml")
	if err := viper.ReadInConfig(); err != nil {
		panic(fmt.Errorf("read config file error: %s \n", err.Error()))
	}

	// 将配置信息反序列化到结构体中
	if err := viper.Unmarshal(&C); err != nil {
		panic(fmt.Errorf("unmarshal config error: %s \n", err.Error()))
	}
}

func GetConfig() *Config {
	return C
}
